from airtest.core.api import *
from poco.drivers.android.uiautomation import AndroidUiautomationPoco

poco = AndroidUiautomationPoco(use_airtest_input=True, screenshot_each_action=False)


# ---------------------------------------------------------华为P50-------------------------------------------------
# 华为 P50 相机图标元素定位
Machine_icon = poco("托盘,相机支持服务卡片")

# 华为 P50 相册图标元素定位
Comes_photo_album_huawei = poco("托盘,图库支持服务卡片")

# 华为 P50 录像模式按钮元素定位
Photograph = poco("录像")

# 华为 P50 拍照模式按钮元素定位
local_photo_button = poco("拍照")

# 华为 P50 获取帧率文本需要用到的元素定位
fps_show_text = poco("com.huawei.camera:id/fps_show_text")

# 华为 P50 获取分辨率文本需要用到的元素定位
size_show_text = poco("com.huawei.camera:id/size_show_text")

# 华为 P50 拍摄快门按钮元素定位
shutter_button = poco("com.huawei.camera:id/shutter_button")

# 华为 P50 用户获取当前相机分辨率及帧率的父级元素定位
fps_elenment = poco("android.widget.FrameLayout").offspring("android.widget.LinearLayout")

# 华为 P50 拍摄分辨率文案节点元素定位
comes_with_camera_clarity = poco("com.huawei.camera:id/size_show_text")

# 华为 P50 拍摄清晰度文案节点元素定位
comes_with_camera_frame_rate = poco("com.huawei.camera:id/fps_show_text")

# 华为 P50 所有照片相册按钮元素定位
Comes_photo_album_all_photos = poco("android.widget.FrameLayout").offspring("本地相册").child("android.widget.RelativeLayout")

# 华为 P50 相册中首个视频文本时间获取元素定位
Comes_photo_album_time = poco("android.widget.FrameLayout").offspring("androidx.recyclerview.widget.RecyclerView").child("android.widget.RelativeLayout")

# 华为 P50 相册选择视频元素节点，后面跟索引
Comes_photo_album_select = poco("android.widget.FrameLayout").offspring("androidx.recyclerview.widget.RecyclerView").child("android.widget.RelativeLayout")
# ---------------------------------------------------------vivoS7-------------------------------------------------

# vivo S7 相机图标元素定位
Machine_icon_vivo = poco("相机")

# vivo S7 相册图标元素定位
Comes_photo_album_vivo = poco("相册")

# vivo S7 录像模式按钮元素定位
Photograph_vivo = poco("录像")

# vivo S7 拍照模式按钮元素定位
local_photo_button_vivo = poco("拍照")

# vivo S7 设置按钮节点元素定位
Photograph_set_up_vivo = poco("com.android.camera:id/camera_setting_button")

# vivo S7 获取帧率文本需要用到的元素定位
fps_show_text_vivo = poco(text="帧率")

# vivo S7 获取分辨率文本需要用到的元素定位
size_show_text_vivo = poco(text="录像分辨率")

# vivo S7 30 帧节点元素定位
fps_show_30_vivo = poco(text="30fps")

# vivo S7 60 帧节点元素定位
fps_show_60_vivo = poco(text="60fps")

# vivo S7 4K 分辨率节点元素定位
size_show_4K_vivo = poco(text="4K")

# vivo S7 1080p 分辨率节点元素定位
size_show_1080p_vivo = poco(text="1080p")

# vivo S7 720p 分辨率节点元素定位
size_show_720p_vivo = poco(text="720p")

# 设置返回按钮节点
Photograph_set_up_back_vivo = poco("com.android.camera:id/main_settings_title_back")

# vivo S7 拍摄快门按钮元素定位
shutter_button_vivo = poco("com.android.camera:id/shutter_button")

# vivo S7 相册全部项目按钮元素定位
Comes_photo_album_all_photos_vivo = poco("android.widget.FrameLayout").offspring("com.vivo.gallery:id/gallery_root_bottom").offspring("com.vivo.gallery:id/rv_albumset").child("android.widget.LinearLayout")

# vivo S7 第一张视频节点元素定位
Comes_photo_album_frist_video_vivo = poco("com.vivo.gallery:id/gl_root_view")

# vivo S7 设置节点元素定位
Comes_photo_album_set_up_vivo = poco("详细信息")

# vivo S7 相册中首个视频文本时间获取元素定位
Comes_photo_album_time_vivo = poco("com.vivo.gallery:id/value_video_duration")